#pragma once

#include <QTreeView>
#include <QVector>
#include <QVariantMap>

// 前向声明
class TreeModel;

// 自定义树视图
class DragView : public QTreeView {
    Q_OBJECT

public:
    explicit DragView(QWidget *parent = nullptr);

    void setTreeModel(TreeModel* m);

protected:
    void startDrag(Qt::DropActions supportedActions) override;

private:
    TreeModel *model;
};
